[设计模式]之十:组合模式

定义

将对象组合成树形结构以表示“部分-整体”的层次结构。组合模式使得用户对单个对象和组合对象的使用具有一致性。

UML

示例代码

// 声明组织具有的功能，所有子组织都要实现这些方法
abstract public class Component {
    protected String name;

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public abstract void add(Component c);
    public abstract void remove(Component c);
    public abstract void display(int depth);

}
// Leaf节点，即实现部门功能。
public class Leaf extends Component {

    public Leaf(String name) {
        this.name = name;
    }

    @Override
    public void add(Component c) {
        System.out.println("Leaf cannot add anything");
    }

    @Override
    public void remove(Component c) {
        System.out.println("Leaf has nothing to remove");
    }

    @Override
    public void display(int depth) {
        String str = "";
        for (int i = 0; i < depth; i++) {
            str += "-";
        }
        System.out.println(str + name);
    }

}
// 组织结构，可以添加子组织
public class Composite extends Component {

    private List children = new ArrayList<>();

    public Composite(String str) {
        this.name = str;
    }

    @Override
    public void add(Component c) {
        children.add(c);
    }

    @Override
    public void remove(Component c) {
        children.remove(c);
    }

    @Override
    public void display(int depth) {
        String str = "";
        for (int i = 0; i < depth; i++) {
            str += "-";
        }
        System.out.println(str + name);

        for(Component c : children) {
            c.display(depth + 2);
        }
    }

}
//客户代码
public static void main(String[] args) {
    // TODO Auto-generated method stub
    Composite root = new Composite("总公司");
    root.add(new Leaf("财务部"));
    root.add(new Leaf("组织部"));

    Composite comp = new Composite("深圳分公司");
    comp.add(new Leaf("财务部"));
    comp.add(new Leaf("组织部"));

    root.add(comp);

    Composite comp2 = new Composite("香港分公司");
    comp2.add(new Leaf("财务部"));
    comp2.add(new Leaf("组织部"));

    root.add(comp2);
    root.display(1);
}
//输出
-总公司
---财务部
---组织部
---深圳分公司
-----财务部
-----组织部
---香港分公司
-----财务部
-----组织部

评价

需求是体现部分与整体层次的结构时，如果希望用户可以忽略组合对象与单个对象的不同，统一地使用组合结构中的所有对象时，就该考虑用组合模式。

组合模式定义了包含子部门的基本对象和分公司这样的组合对象的类层次结构。基本对象可以组合到组合对象中去，组合对象又可以被组合，这样不断递归。